# 3.开房查询，从控制台输入名字，查询在kaifanglist.txt文件中的开房记录，如果没有，是一个单纯哥们，如果有的话，将其所有开房信息写入到以这哥们命名的文件中
def read_data():
    with open('data/kaifanglist.txt','r',encoding='utf-8') as f1:
        record = f1.readlines()
        print(record)
    return record


def write_data(name,name_list):
    #因为这里是在同一个资源写入,所以不需要'a'追加

    if name_list:
        print(f"{name}果然去开房了~~~")
        # 查询到了数据
        with open(f'data/{name}.txt', 'w', encoding='utf-8',newline='') as f2:
                f2.writelines([line + '\n' for line in name_list])
                f2.flush()
        print(f'{name}的数据提取成功')
    else:
        # 未查询到数据
        print(f"未查询到{name}相关的数据，{name}是一个好哥们")

def search_data(name):
    data = read_data()
    name_list = []
    for item in data:
        infolist = item.rstrip(",\n").split(",")
        item = item.rstrip(",\n")
        if infolist[0] == name:
            name_list.append(item)
    print(name_list)
    write_data(name,name_list)



if __name__ == '__main__':
    search_data("孙旸")


